import os
from dotenv import load_dotenv
from openai import OpenAI
load_dotenv(dotenv_path='D:\桌面\Project\PythonProject2\.env')


# 1. 先手动定义密钥（测试用，后续可替换为环境变量）
MY_API_KEY = os.environ.get('DEEPSEEK_API_KEY')  # 务必替换成你的实际密钥
# 2. 验证密钥是否有效（打印前3位和后3位，避免泄露完整密钥）
if MY_API_KEY:
    print(f"密钥已读取（部分显示）：{MY_API_KEY[:3]}...{MY_API_KEY[-3:]}")
else:
    print("错误：密钥为空，请检查是否填写正确！")
    exit()  # 若为空则直接退出

# 3. 用验证过的密钥初始化客户端
client = OpenAI(
    api_key=MY_API_KEY,  # 直接使用上面定义的密钥
    base_url="https://api.deepseek.com/v1"  # 固定的DeepSeek接口地址
)

# 4. 简单测试调用（可选）
try:
    response = client.chat.completions.create(
        model="deepseek-chat",
        messages=[{"role": "user", "content": "测试连接"}]
    )
    print("调用成功！返回结果：", response.choices[0].message.content)
except Exception as e:
    print("调用失败：", e)